Upload code and figures

Commit of my work for first peer assesment