Set also xlim to the size of the input in pcolormesh

I see you are only setting `ylim` and not `xlim`, is there a reason for that?

I did a few heatmaps and they had annoying right side white space due to that, so I suggest to change it.