Add graphics/improve visuals