Fix for hw2-1