Update classes to accept game object as parameter