Adding tests for --balance and --weight-field