MSBA_Exercise_1_solution_v2.py