Missing comments added for any features
Remove unnecesary prints of the model.