Adding more tests for main.