copy code over
change the tests