fixed bug and improved usage
fixed bug and improved usage
pep 8