Create snake part 2.py