Make the code actually work