working on pygame animations