Correção, troca if por while em list