Issue making 3D and 2D plots in the same figure using matplotlib

[vidarsko]

The fact that 3D plotting requires 3D axes objects and 2D requires normal axes, it sometimes leads to trouble when generating subplots. Maybe we could implement a check in the plotting functions that checks if the axes are of the correct type and changes them if not.