Step2: Updated scaffold to reflect book example