Same with this one. Should be an example for it and it should also go to the notebook later on. Maybe have a seperate notebook for regression at some point