a bit better UI
Fix fix