Tweak CSS plugin code