Tweak to load balancing
Fixed spatial domain decomp with angular enabled