Add class existance check