Worked on a little UI stuff