moving test code to source folder