Added AI file handling