Adding ui code