Add coments to the code