Put tests in separate file