HW3 part2: fixed issue with div