Updated logic to add missing file if needed
Merge pull request #6 from Scrier/shell-updates

Updated logic to add missing file if needed