complete HW1, -extra credit