Added lift function to clean up distill example