HW2
working imshow