Added WIP class diagram