corrigindo erro no codigo e melhorando um pouco a organização