Fixed problem with amplitudes in find_phase method