* There are still some layers we have not added. * Remove a few layers since the actual model is too big for us to train and evaluate.