- Update of modules - Use of categorical model instead of binary class - Change last activation function to Softmax instead of Sigmoid