closed #4 - update class diagram