fixing code to work on server
Merge pull request #3 from matthew-chell/dev

fixing code to work on server