Quick and dirty implementation of all models that we probably need