code updates + some plots