erectile disfunction

Im 3 months post radical open prostatectomy i had nerve sparing on both sides and when the catheter was taken out i’ve been continent since and cancer free at my first PSA test. ED is a problem though medication hasnt really helped or a pump is injections the next stage?