add some button listener code