small class change