Create get_retention.py