Created by: Rikuoja
Now all models should work as intended. Did some fixes in the classification functions too.