NBC: Ron Klain is expected to step down as White House chief of staff in the coming weeks