correct types
update docs