fixed up the homework viewer