Improved config generator asking for stages and config