sweet pull request, bro

Just some minor updates :)