trying to fix part 2