Little fixes

My first pull request, yay me :)