First working version with two buttons