Tweaks, document more code