Added test in main.