Added bug fix/error checking for generating the html. Also added in feature request for Homework 6.