Fix used package name

Fixes #1

Also adds missing doc