Add Empty Dimension To Numpy Function Python