new function to parse the training function

I created a new function which will return the cell of 40 optimised
function for each layer
Merge branch 'master' of https://github.com/ChiminhTT/ML_CW2