Added add button logic