Resolves #1.
Updated the plugin so that it can take a controller as a parameter. Also re-ordered the parameters to move controller before type, as it's more important.