Added button feedback logic