Ui improve

Try it out. See if you like it or not