Added logging option. Added more plots.