Made some small changes, now includes muons in track collection