Update management.py