Add run_at document_start