Write some more docs and tweak some stuff