The code is fixed, and it's aligned with the second assignment's requirments