Fixed import statement
Removed pprint import