Updated action function signature