Fixed two small bugs
Merge pull request #11 from fredmcfly/patch-1

Fixed two small bugs