forgot to add the test file. minor fixes included