Clean up code
AI can calculate winning moves in the vertical direction