refactor color jitter