Balance unit strength; update TODO