[WIP] Update argument format in the docs