fix(ui): Add thinking screen

This should have never gone into this branch anyways...