Updated function signature