changed the test
moved the code from test to src