Updates for additional functionality

Sorted out plotting some histograms