Decided that the plugin was a little convoluted and didn't want to have to re-edit it each time it gets updated. Removed that and added a much simpler function to functions.php instead.