Add explicit return to main