spruced up ex1 figures; updated TODO