updated it to actually work