Add class variable
Update test