input: Write a python function to search for a query in a list and return the results.
  assert search([1, 2, 3, 4, 5, 6], 4) == [4]
  assert search([1, 2, 2, 4, 2], 2) == [2, 2, 2]
  assert search([1, 2, 3, 4, 5], 6) == []
output: def search(lst, query):
      return [item for item in lst if item == query]