Updated the code to address the peak and current magnitudes in the cases where the probability is that the SN is not a Ia