Add data manipulation and linear regression