Create snake part 4.py