touching up code
Merge pull request #19 from palmersndrsn/touchups

touching up code