Small update

Change to LinearElement classes