Change return value in main