Added basic graphics code