Refactoring to properly use pygame sprites.