Add command line options for running subsets of stages