Updated with required css class