Fixed the code from my last commit, all buttons work now