Code Update

Work on AI