HW 3 part 3 fixed restler issue