-removing todo as unit test can't test anything
-changed imports