Added documentation and more plotting options to equilibrium.py