Polynomial Basis Expansion for Feature Engineering - Coding problem | OneNoughtOne