rework the main proof
small notation change