ADD: Part 2 py file for notebook