<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def square_in_front(self, location=None):
 """ Finds square directly in front of Pawn :type: location: Location :rtype: Location """

        location = location or self.location
        return location.shift_up() if self.color == color.white else location.shift_down()