Updated model running demo to be awesome-er.