Put tests in main