Done, I left out the make_square method though, because it really only works for contact maps as written... So it probably only makes sense to call it as md.geometry.contact.make_square. Or I could rename it I guess