Setup instead of fit