Modifications to AI. Hard is almost ready