Task 1 (Stochastic PI)
Task 2 (Softmax Regression)