Added code to Part B section