Added functionality for add button.