Add script for calculating amplitudes