added missing end
Merge pull request #7 from Ch4s3/add_capacity

added missing end