minor changes + flow of sample run