adding code part 2