updated the ui to make the flow a little easier