Fixed main class diagram