Changed AI package structure