arrumando um pouco a bagunça do codigo adicionando algumas funcoes