Updated with final comments - ready for submission to Coursera