Make graphics its own class