Working as intended! I should probably refactor, though.