Added return to the GameplayScreen after saving game
Also added the class diagram