Adding favorite logic