added label dist visualization code, validation set creationg code