implement a way to track usage of different models
implement a way to track usage of different models