Moved unit test call to main