Updated class diagram and screenshot.