Setting correct version and adding import statements

Polishing the docs