added hmwk 2