Use class instead of ID