AI hw4 getting ready for submit