added data generating code for final project