Add the source code
Minor changes