Modified Attention graph