main.c: Corrige lógica e adiciona opção -r

Signed-off-by: Aryane Ast dos Santos <aryaneast@gmail.com>