Update to paper after Ewa's comments