Testing suite and improved AI