adding animation with pygame