Update README with instructions
Add visit wrapper logic
Class cleanup