my try at the HW/ lab

looks like all the tests are working