Updated docs, small tweaks