Add all the codes linked to the sgd model