Incorporated command in main flow

Incorporated command in main flow