A plan to increase tax on large super balances has been overhauled, with Labor saying they took feedback on its original plan seriously.