Adding a few more methods and updating batch sizes