Adds plotting function
Adding writeup