wrote tests, must modify to pass tests