Tweaks to code toggle styling