Tidy up slightly FE total Lagrangian scheme

Tracker item: 3675