Add a couple more UI elements