moved the code to test file