batch size parameter added