Improve stats ui