diff --git a/basic_management.py b/basic_management.py
index 298ea49..49e877c 100644
--- a/basic_management.py
+++ b/basic_management.py
@@ -1,6 +1,6 @@
-from subprocess import getoutput
-import docker_parser
+import docker_wrapper
 import pprint
 
 # Goals: Loop: Select container (or exit), then menu to do stuff to container (start, stop, rm, list info, go back to main menu)
-pprint.pprint(docker_parser.DockerParser.ps())
+pprint.pprint(docker_wrapper.Docker.ps())
+pprint.pprint(docker_wrapper.Docker.containers_info())