Adding button push logic