Add support for 3D and 1D data

[koriavinash1]

Most of the current functions knowingly or unknowingly are inclined towards 2D image data, need to generalize this