Trying to fix part 2