Write a Python function that takes a list of words as input and capitalizes the first character of each element in the list.
["apple", "banana", "strawberry"]
def capitalize(words):
  return [word.capitalize() for word in words]

input_words = ["apple", "banana", "strawberry"]
capitalized_words = capitalize(input_words)
print(capitalized_words)