little changes
Merge pull request #4 from rphitt1/master

little changes