Working on small dataset

Coming soon.