fixing the repo
small dataset for visual example