Writing more tests, currently working on testing assign_batteries