Classic snake game in python