Updating the hook to actually work I think