Add shebang support