Minor improvements

Hey Bryce,

I like how simple this plugin is and I decided to (hopefully) make it little bit better.

Lemme know what you think.

Cheers