Python code for polynomial regression