Homework for IMP6 class