function $ (query) {
  var e = document.querySelectorAll(query)
  return e.length === 1 ? e[0] : e
}