Adding more UI