Transfer matrix simple code