Add Code for LIREG and NN