Using commandline arguments to pass directory names

Something I felt would be nice to have, so that people who use it don't have to edit the script.
`PEP8 Error: E225 missing whitespace around operator`