Found the if/else that I forgot to change. Everything seems to be working. Need to make some changes to the UI and add some effects for fun