fixing batch size