add colorization
checkpoint