New code for 2 buttons