Added class diagram to implementation