Fix main file references
Merge pull request #121 from willcosgrove/patch-1

Fix main file references