The correct code for the photo album editor