change working dir to current dir. Small PEP8 changes