Just a simple snake game in Python