HW2 #3 ready but not beautiful