Minor tweaks
Merge pull request #2 from milikicn/patch-2

Minor tweaks