Numerical model of electro-osmotic consolidation in clay