updates to buttons and code