Whoops. Add shebang
Fix import