Refactor unit terrain stats