Copied main over and updated sample