そのままだと、「ざんねん！」の時だけしかメインに戻らないので、IF文の後ろではなく、行を改めて次の行でRETURNを入れてください。