Minor update on part 2