Tighten up returns from check_for_end()
Changes tests to match
Add AI move tests