Add files to repo

Adding figures in /figure folder and code files to repo