Prims Code In Python For Games