Added input initialization as an argument for Gaussian mixture model