Add alternative implementation for reg4_em_step, which takes advantage of matrix operations and uses a different set S
Change default regularizer (since we are now dealing with 3d) and change initial and final temparature to better match the current our data scaling