Add picture to README.md. Further modularizing code.