added missing part of the assignment