Basic python game using pygame