add possibility use for different models