Make use of stage specific branch option

Issue #77 

Use the branch option of the stage if it exists