PS2 - Regression code