small tweak, still need to find a way to ensure that optical flow is correct.
small tweaks.