Changes mades to attempt at third step of coursera