initial code part 2