Implented UI with multiple stages