included proba estimation in agent and create first test code