update dimuon code