Clean up put statements.

Just clean up code so it doesn't print a long list of your files. I also
realised it was easier to just hard code the directory. Will need to add
support to change this later.