To run this project:

mjpython multi_reward_elegans_demo.py --episodes 1 --num_rewards 2

Change episodes and number of rewards as you want.