python done
tweak the code