Added code to distinguish between decays