Make the code actually work \o/.