code cleanup for dataset + readme