Added function to plot Motility classes.
Refactored the sweep to pass classes plot.