Modify the screen UI for stt mode and refactoring code