Fix: update README, better import statement