styling and better UI
