Refactored reg flows to accept a broader range of args