updated ui code