Update for assignment on coursera