Fixed add functionality