Problems in week1 assignment2

Tom has already pointed out the problem here, but maybe it’s also worth pointing to this Discourse post which is a much more complete introduction to the Sequential and Functional APIs than what they give us in the notebooks.