Updated class diagram and screenshot