Changed import to work with Pillow (PIL fork)