Stage 2 with comments

added some comments