setup class from previous code