Actually updating the code here

I should've done this every time I've done it in the master branch