Snake implementation using Python and Pygame