Fix nits
Merge pull request #2 from dankurka/master

Fix nits