Fixed UI look a little