A simple game in python with pygame