Improve AI spell casting