Updated code for Geoffrey's AI

2s will now be in the correct positions.