fix notes from review