Add hashbang for python, import cleanup