put some comments from the paper on the code