Code for social cross-situational learning experiment