Implement tree operations; convert to lenses
Adjust styles a bit