ENH: improved handling of less common data situations
TST: added test for Heckman MLE fitting
TST: added test for handling missing data