Added snapshot function for running model