fix(ui): Add thinking screen