Building a snake game using pygame in python