Update docs

Updated to match some API changes mostly