Update to make graphics work