changed main back to test cases