HW 3
updated