Updating UI to reflect the clean flow with a header