Add aspect argument to imshow

The motivation for this is that I'm plotting the energy deposition density across a square detector, and want to be able to pass `aspect='equal'` so that the aspect ratio of the plot matches that of the physical detector.