Clean up (wage equation instead of Heckman)
Bugfix