L4 Assignment done, have to make sure the tests work