Added a return to make the plugin chainable