Adds fix for heat directory bug
Merge pull request #88 from trobert2/bug_repair

Adds fix for heat directory bug