Support multiple stages

R: @brymaven