Add ability for heat_scatter_3D to return figure handle.