Update docs, fix func