Updated repo to reflect the tests from last assignment