small change, better UI