Create custom_heuristic.py