Tiny changes for my first pull request ;)