Add support for choice parameter