Made handling the different models a bit more convenient