added batch size