Cater for warnings
Merge pull request #5 from jeffmess/fix-warnings

Cater for warnings