Updated UML diagram to be a StateBasedGame