Fix some bugs.
Fix bug in plotting mesh and save .png file in the images folder.