Addresses issue #150
@jspah can you please review this MR since you have muon experience from your analysis example as well as the base.py experience?