Working on tests for Fundamental Transformer