fixes capacity
removes dummy variable
Merge pull request #60 from andreywix/Feature-PolishingUp

Feature: Polishing Up