Add support for selectKey and generated keys in update statements