Updating the visuals to fix a couple bugs.