Minor improvement on focus stacking