Updated parameters to be per class