You should be able to add a UI step in `subiquity/ui/views/`. Create a `desktop.py` file and then implement the logic.