Simple 2D game in python using pygame