Added a flag to concatenate jet features with tracks representation after Transformer step. Relates to #16 (closed)