small changes with buttons