AI2 assignment 5 initial