monkey_model.py maybe is working now
Updated notebook with first test with the new code