Up to problem 4

Separated everything in modules. Calculated choice specific values and
switching costs. Plot too.