Debbuging and improving AI, still not finished