How would you modify the linear regression TensorFlow code to accommodate an input of 4 features instead of just 1?