Made minor changes to use facebook dataset.