Simplify class diagram