Julia's review

Including minor update to figures