Edit the following code snippet to calculate average in Python.
def calculate_avg(values):
  total = 0
def calculate_avg(values):
  total = 0
  count = len(values)
 
  for value in values:
    total += value
 
  return total/count