made the grading driver work for week-3-assignment-2