Code with comments added