#3 Very small update to class diagram