Add support for setting documents IDs

OTS: Guy
TESTED: unit tests and AWS