Style fixes + print function

- Imported print function
- Removed unnecessary parens from from if
- Removed unused import