Polishing

Some polishing (added comments, changed way of referring to data frame)