Add set_geometry to DataFrame as well