Add styling and ux