Adding puzzle with solutions
Fixed some bugs and added comments