Add handleClass parameter