Updated the feature based on new platform.

Signed-off-by: Staris <jaison@worxco.com>