Adding equity.py