Adding YAML to homework check script