adding the ui code