added functionality to add button almost