Some improvements in lmms_math