added additional_unit and ability_upgrade method calls

removed space

Fixed return type