Update take_image.py