Updated comments - final version for submitting to Coursera