Made minor tweaks to overlay plugin