Add option to turn off stage1