update homework 11 class diagram