#6 passing all tests given in assignment, but not finished yet