Add legend
Add TODO
Organize notes