AI course assignment 3 - BT tree