another fix for lab 6 week 6