Added initial AI implementation (Not so good though)