Changed constructor example to include the use_stage parameter