Even though it appears that our little solution is working correctly, we still have to keep in mind that the model has been trained on a very small amount of data. 

To improve the model, from the Default Custom Model page, you can click on the blue button labeled Edit and Retrain:

This will make our project editable.