Updated the code to actually work (broke it by mistake).