Added decays and fixed saving bug