Added shebang and removed unnecessary import.