Update median.py