small changes to pygame lab