Improve the UI. Needs more features.