make minor changes from solution vid.