Add code sample for part 2