Implementing AI in my problem solving!