fix return type of main