Fix for Homework 9