Add sole
Fix the averaging in bench