Moving tests to src directory