This is the correct code:

  def formatted_zip_code(digits)
    case digits.size 
    when 4 then "0#{digits}" 
    when 3 then "00#{digits}" 
    # 2 or less digits is an invalid input
    else digits 
    end
  end