update to actually work