Generalize figure helper functions

[aarmey]

Functions like plot_ligands and plot_cells should not care about which figure they are being plotted on. These if/else statements should be removed, and the arguments changed so that they are generic.