Add more changes to plugin