improved pep8 and corected as Lucas sugested