Quick and dirty prototype
Add tokens color
Fix scrolling