Replaced flexbox with 2x2 matrix