Updated codes and plots