Improve stats visualization
Improve UI