Improve stats UI