Python Minimum Spanning Tree