week 6 - class diagram improvements