diff --git a/README.md b/README.md
index 3f3f8c6..aa2cda7 100644
--- a/README.md
+++ b/README.md
@@ -45,7 +45,7 @@ To load and draw the HOMO:
 
 ```python
 from yutility import orbitals
-import tcviewer
+from tcviewer import Screen, materials
 import pathlib
 
 rkf_dir = pathlib.Path(__file__).parents[0]/'data'/'NH3BH3'
@@ -56,6 +56,6 @@ homo = orbs.mos['HOMO']
 # generate a cube file
 cub = homo.generate_orbital()
 
-with tcviewer.Screen() as scr:
-	scr.draw_cub(cub, 0.03, material=tcviewer.materials.orbital_matte)
+with Screen() as scr:
+	scr.draw_cub(cub, 0.03, material=materials.orbital_matte)
 ```