Added the facility of batch size. More flexibility!