Adding feature to edit photos