Updated CNN training app