add arguments for model