Klaus Mann, Schriftsteller