Added tests to test/main.