Added axes parameter to heatmap to allow for imbedding into e.g. gridspec layouts.
Added axis parameter to trajectory plotting functions