some small changes after finishing Prims.