Final product for yei
Added button instead of text