Mi sembra di parlare con un muro di gomma.
Non puoi modificare la funzione. Quella va bene così.
Devi cambiare la chiamata nel main e fare la if nel main. Niente altro.
Più di questo, che suggerimento vuoi ? Ti devo scrivere le modifiche del codice al main ?