Update to do, remove class diagram