Updated plugin to be more wordpress based