Added test for main function (doesn't pass yet due to stack overflow)