Deletes some comments from code. Adds images for the paper