Squared Attention - Compositional Interpretability