Little modification in logic flowchart