Add episodes/season back in refactor code