Don't need to restrict omega for parallel SGD
Update profiling