Update album remove function