Improved UX for the amb example