Trying to fix GUI updating