A 2d game in python