Update 2d_to_3d.py