Add more UI methods