forgot to add changed code too

for the new graphics