complete comments and finish HW2!