Investigating the role of model-based learning in exploration and transfer