Add support for the new SET command