Added plot and code