ECE-575 - Adding Batch Feature