Updated script + added error on training and dev