would you want to create a pull-request?

I edited the code on github, but wasn't able to test the changes there. They're simple adjustments though, and I tried not to break anything so I think it should be ok.