Change source to source_dir
Merge pull request #2 from hatchan/patch-1

Change source to source_dir