Added AI.
Plus some other features