Add some plot formating for the matrix plot