Added the final code changes needed to complete design lab #3