Realized I was dumb and made the code harder than it had to be. Now its
simple.