Improves the UI