updated linear units code