Make changes to attempt to update plugin to work with Wordpress 4