Add AITest with most methods and clean up AI