Removed unnecessary PIL import