Added layout for part 2