change function declaration to fit the new way to do it