improve the ui styling