added code for exploration 2