Fixing a bug in adam's week 2 code.