Game in python using pygame