Last minute bug fix for Homework 6 Demo