Updated
Added figures and code to appendix