adding some ui stuffs