migrated UI to Pygame