small changes on ui