did a couple of todo notes
minor update