Add basic graphics handling