cleaned up and made a couple fixes
Merge pull request #6 from b91961/master

cleaned up and made a couple fixes