Update docs, unit tests