Did some more ui work.