Multiple Main Stage Functionality Implementation

Implemented functionality for addition of multiple main stages for workload config UI