Trying to fix graphics