fixes issue #5
Merge pull request #8 from kirsty-hames/issue/#5

fixes issue #5