Fix add() method