Replace contain with class