Simple game written in python using pygame