Add class to images