fixed add module .py issue and added ability to add a header to files