Minor start on class diagram