Use data directory for the data files
Add extra comments to code
Add .png plot files