I have nearly no time to work on it, but I fixed a few things on the train: (dummy data sets )