Editable drag behavior on 3d plot

So, I’m trying to make a 3d plot (scatter plot is enough) where the user can drag and edit/move points with mouse. I got this nice help from here http://bl.ocks.org/niclasmattsson/7bceb05fba6c71c78d507adae3d29417 which can be used perfectly for 2d plots. Now how can I implement that for 3d? I just want to move the points in z direction in a 3d scatter plot. Can that be done with plotly?

It should be doable using a combination of plotly_click and Plotly.restyle.