Register each stage so a single example can run
Register each stage so a single example can run