Cleanup main function
Unit tests are run before main program starts