Optimize the solution of the Fish challenge
Add some comments, and remove unnecessary code
Remove duplicate code