Add code from Polina's presentation.