The code works... for now