moving test to Main.