Tweaks after writing docs