Update profiling.py