Added some graphics and UI