Add lookup function to the base model, this needs to be more high level