Adding a bit of code for the eventual button transition.