Added plot scripts and make sure it runs