Improved the comments
Added a threshold to avoid the free surface artifacts
Changed the configurations