cs501-hw1-dataset