Adding in some UI and made the initiation flow a little better