Updated initial version of class diagram