[ENH] UFEM: Improve assembly benchmarks