add some simple solution and comment to help understand the code