Recover lost UI

Still working on it.