HW4b, but needs a lot of imagination more