A correlated basis set for nonadiabatic energy calculations on diatomic molecules