My version of snake implemented with pygame