Update and rename test.py to main.py

create two button