GitHub repository: pvsundar/pymarginaleffects
Description: small correction to the replication Code