Prediction of erectile function following treatment for prostate cancer.