Done with search, just missing heuristic