Add extra loop for jit optimization
Change name of function to update positions of all particles