Como criar uma função que retorne o elemento em JS

function elem(did) {
  return document.getElementById(did);
}
register_fm = elem('register_fm');
//---------------OR---------
function elem(did) {
  return document.querySelector(did);
}
register_fm = elem('#register_fm');
register_fm = elem('.register_fm');
register_fm = elem('p');
register_fm = elem('div');
Dan George