* added logic to comp rays/s on train and vis * added test and fixed vis rays/sec * added flag to just visualize dataset * added new script for viewing only dataset