Implemented model for beluga estimator in beluga.py