playing with an Elastic net model, and added a few more features