Limpei um pouco o código