adds better UI