Created separate .py file to generate 2D/3D social graph