Add class instead of safe attribute