working on dataset creation
committing with MLP