Added class diagram (WIP)