Add code

Add code for part1