Small improvements on writeup