Add tests for labeling, plotting the actual heatmap, and setting the title
Set defaults to plot as None for everything
finished tests?