ah! UI Fixes