Add shebang to python