A game of snake on the command line in python