Adding comments and implementing changes after code review