add axis argument to function.product()