added TODO and made hardcoded grid smaller