Add class diagram. Update Related work and Lessons learned in word doc.