HW4 part2 fix
Moved images