Add harmonic oscillator potential