fixed a couple of bugs on add