Basic Snake - Training with pygame