Edited test_main