Parameters and reference

Added the extra parameters of the function and added the reference to the arxiv paper.