Add #!/usr/bin/env python to test