Fixed return value in main