add some button stuff