Diagnose and fix linear regression assumption breaks | Citibank Interview Question