Added functionality to add button