Fixed plotting. Makes heat maps properly.