Working on the A* code