Add easier modalities plotting and calculation