Updating some code to actually work