Updated AI trains and applied minor fixes