Got add button fully working