Updated to use new graphics module.