Followed up on code review