Mole has AI, code cleanup