Take out setCapacity, made only to test methods at the beginning
Improve a comment in main