Added source code in writeup sheet4