Fix return value in main function