Rewrote collection display. Used collection view. Now it works automatically.
Issue #25
Issue #23