switch to nx.spring_layout