Ruben Östlund adapted for the stage by Tim Price