Updates to plugin