Added graphics code