Jessica Chastain returns to theater with ‘A Doll’s House’