Added comments for Stage 4