clarify 2x2 case and add figures