Adding ui changes