Fixed bug in exploration.py