ML assignment 5 polynomial features