Tag snake game in python