Jude Law returns to the stage in Obsession