```ruby
module Imdb
  class Season
    attr_accessor :id, :url, :season_number, :episodes

    def initialize(url)
      @url = url
      @season_number = @url.scan(/episodes\?season=(\d+)/).flatten.first.to_i
      @episodes = []
    end

    def episode(number)
      episodes.fetch(number-1, nil)
    end

    def episodes
      @episodes = []

      document.search("div.eplist div[@itemprop*='episode']").each do |div|
        link = div.search("a[@itemprop*='name']").first
        @episodes << Imdb::Episode.new(
          link[:href].scan(/\d+/).first,
          @season_number,
          div.search("meta[@itemprop*='episodeNumber']").first[:content].to_i,
          link.content.strip
        )
      end

      @episodes
    end

    private

    def document
      @document ||= Nokogiri::HTML(open(@url))
    end
  end
end
```

Fix the .episode() method too.
user
```ruby
module Imdb
  class Season
    attr_accessor :id, :url, :season_number, :episodes

    def initialize(url)
      @url = url
      @season_number = @url.scan(/episodes\?season=(\d+)/).flatten.first.to_i
      @episodes = []
    end

    def episode(number)
      i = episodes.index{|ep| ep.episode == number}
      (i.nil? ? nil : episodes[i])
    end

    def episodes
      @episodes = []

      document.search("div.eplist div[@itemprop*='episode']").each do |div|
        link = div.search("a[@itemprop*='name']").first
        @episodes << Imdb::Episode.new(
          link[:href].scan(/\d+/).first,
          @season_number,
          div.search("meta[@itemprop*='episodeNumber']").first[:content].to_i,
          link.content.strip
        )
      end

      @episodes
    end

    private

    def document
      @document ||= Nokogiri::HTML(open(@url))
    end
  end
end
```
assistant