Added main to run unit test.