Python project for linear regression