Just wanted to update a few things

Hey-o
Came across your plugin and wanted to use in a project so I made a few updates and wanted to share. 
Feel free to merge or not, totally your call.