more ui improvements, hadn't had a chance to look at it though