minor fix
pep clean p
new code