fixing final question for assignment