Updated Sponge implementation