An axisymmetric gravitational collapse code