Added some graphics and fixed some problems