changing batch size from 16 to 8