Updated the feature based on the new platform.

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